package org.eda1.practica01.ejercicio02;

import java.util.ArrayList;

public class EmpresaProyectos {

	private String empresa;
	private ArrayList<ProyectoCiudades> proyectoCiudades;

	public EmpresaProyectos() {

		this.empresa = "";
		proyectoCiudades = new ArrayList<ProyectoCiudades>();

	}

	public EmpresaProyectos(String empresa) {
		this.empresa = empresa;
		proyectoCiudades = new ArrayList<ProyectoCiudades>();
	}

	public void addProyectoCiudad(String proyecto, String ciudad) {
		boolean proyectoCreado = false;

		for (ProyectoCiudades iterador2 : proyectoCiudades) {
			if (iterador2.getProyecto().equals(proyecto)) {
				iterador2.addCiudad(ciudad);
				proyectoCreado = true;
			}
		}
		if (proyectoCreado == false) {
			proyectoCiudades.add(new ProyectoCiudades(proyecto));
			proyectoCiudades.get(proyectoCiudades.size() - 1).addCiudad(ciudad);
		}
	}

	public void setEmpresa(String empresa) {
		this.empresa = empresa;
	}

	public ArrayList<ProyectoCiudades> getProyectoCiudades() {
		return proyectoCiudades;
	}

	public void setProyectoCiudades(ArrayList<ProyectoCiudades> proyectoCiudades) {
		this.proyectoCiudades = proyectoCiudades;
	}

	public String getEmpresa() {
		return empresa;
	}

	public int size() {

		return proyectoCiudades.size();
	}

	public ProyectoCiudades getProyectoCiudades(int i) {

		return proyectoCiudades.get(i);
	}

	@Override
	public String toString() {
		String aux = this.getEmpresa() + ": ";
        for (int i = 0; i < proyectoCiudades.size() - 1; i++) {
                aux += proyectoCiudades.get(i).getProyecto()
                                + proyectoCiudades.get(i).toString() + "; ";
        }
        aux += proyectoCiudades.get(proyectoCiudades.size() - 1)
                        .getProyecto()
                        + proyectoCiudades.get(proyectoCiudades.size() - 1)
                                        .toString() + "\n";
        return aux;
	}
}
